package com.test.ai.utils;

import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.lang3.StringUtils;

/**
 * Created with IntelliJ IDEA.
 *
 * @description:
 * @author: liuziyang
 * @since: 8/21/25 12:04 PM
 * @modifiedBy:
 * @version: 1.0
 */
public class ResponseUtils {
  // 使用非贪婪模式匹配<think>标签及其内容
  private static final String regex = "<think>.*?</think>";

  public static String removeThinkTags(String input) {
    if (StringUtils.isBlank(input)) {
      return StringUtils.EMPTY;
    }
    // 正则表达式说明：
    // - <think> 匹配开始标签（区分大小写）
    // - .*? 非贪婪匹配任意字符（包括换行符，因启用了DOTALL模式）
    // - </think> 匹配结束标签
    // Pattern.DOTALL 使.可以匹配换行符
    Pattern pattern = Pattern.compile(regex, Pattern.DOTALL);
    Matcher matcher = pattern.matcher(input);
    // 替换所有匹配项为空字符串
    return matcher.replaceAll("");
  }
}
